package com.kevin.Code.DP;

/**
 * @author Vinlee Xiao
 * @Classname BestTimeToBuyAndSellStockWithCooldown
 * @Description Leetcode 309. 最佳买卖股票时机含冷冻期 中等难度 动态规划 没啥思路
 * @Date 2021/10/19 19:29
 * @Version 1.0
 */
public class BestTimeToBuyAndSellStockWithCooldown {

    /**
     * 关键理解状态和选择
     *
     * @param prices
     * @return
     */
    public int maxProfit(int[] prices) {

        int len = prices.length;
        if (len == 0) {
            return 0;
        }
        //i天结束之后，持有的股票
        int[][] dp = new int[len][3];

        //表示持有股票
        dp[0][0] = -prices[0];
        //表示不持有股票，且处于冷冻期
        dp[0][1] = 0;
        //表示不持有股票且，不处于冷冻期
        dp[0][2] = 0;

        for (int i = 1; i < len; i++) {

            //持有股票 有两种状态  1.当天买的,前一天不处于冷冻期  2.前一天就持有股票
            dp[i][0] = Math.max(dp[i - 1][2] - prices[i], dp[i - 1][0]);
            //
            dp[i][1] = dp[i - 1][0] + prices[i];
            //i天不持有股票 1.i-1天处于冷冻期  2.i-1天不处于冷冻期
            dp[i][2] = Math.max(dp[i - 1][1], dp[i - 1][2]);


        }

        return Math.max(dp[len - 1][1], dp[len - 1][2]);

    }

}
